#!/bin/bash

# 获取用户输入的显卡型号和数量
read -p "请输入显卡型号: " gpu_model
read -p "请输入显卡数量: " gpu_count

# 获取当前时间，格式为 YYYYMMDDHHMM
current_time=$(date "+%Y%m%d%H%M")

# 准备测试结果文件，文件名包含当前时间
result_file="rocm-${current_time}.txt"
echo "显卡型号: $gpu_model" > $result_file
echo "显卡数量: $gpu_count" >> $result_file
echo "测试项目 | 成绩" >> $result_file

# 定义一个函数来测试不同类型的计算性能
test_performance() {
    local type=$1
    local perf_tool=$2
    local result

    echo "开始 $type 性能测试..."
    # 这里的 perf_tool 应该是执行性能测试的命令或脚本
    # 以下是模拟的测试结果
    result=$($perf_tool)
    echo "$type 性能 | $result" >> $result_file
    echo "$type 性能测试完成"
}

# 示例性能测试命令，需要替换为实际的测试命令
perf_tool_example="echo 123"

# 测试不同的性能指标
test_performance "INT32(TOPS)" $perf_tool_example
test_performance "INT16(TOPS)" $perf_tool_example
test_performance "INT8(TOPS)" $perf_tool_example
test_performance "INT4(TOPS)" $perf_tool_example
test_performance "FP64(TFLOPS)" $perf_tool_example
test_performance "FP32(TFLOPS)" $perf_tool_example
test_performance "TF32(TFLOPS)" $perf_tool_example
test_performance "TF32+(TFLOPS)" $perf_tool_example
test_performance "FP16(TFLOPS)" $perf_tool_example
test_performance "BF16(TFLOPS)" $perf_tool_example

echo "所有测试已完成，结果已保存到 $result_file"